import logging
import torch.nn as nn
from models.hrnet import get_pose_net

class HRNet(nn.Module):
    def __init__(self, cfg):
        super().__init__()
        self.body = get_pose_net(cfg)

    def forward(self, x):
        y = self.body(x)
        return y


def build_intraNet(cfg, **kwargs):
    return HRNet(cfg)